package com.neucloud.da.edgeside.model.enums;

public enum RunEngineEnum {
    TENSORFLOW(1L, "tensorflow","0001.tar"),
    PYTORCH(2L, "pytorch",""),
    SKLEARN(3L, "sklearn","model.joblib"),
    XGBOOST(4L, "xgboost","model.bst"),
    ;

    public final Long code;
    public final String desc;
    public final String fileName;
    RunEngineEnum(Long code, String desc , String fileName){
        this.code = code;
        this.desc = desc;
        this.fileName = fileName;
    }

    public static RunEngineEnum getByCode(Long code) {
        for (RunEngineEnum enumValue : values()) {
            if (enumValue.code.equals(code)) {
                return enumValue;
            }
        }
        return null;
    }


}